#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2024/3/6 15:33
# @Author  : wanghaoran
# @File    : base.py
from abc import ABC, abstractmethod


class BasePromptEngineTool(ABC):
    @abstractmethod
    def get_response(self, query):
        return


class BaseRetrievalEngineTool(ABC):
    @abstractmethod
    # def get_results(self, query, base_name_list, topk, query_vector, embedding_model, similarity_type):
    #     return

    @abstractmethod
    def get_results(self, *args, **kwargs):
        return

class BaseRerankEngineTool(ABC):
    @abstractmethod
    def get_results(self, question, candidate_results, topk):
        return